package com.rd.util;

import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Objects;
import java.util.UUID;

/**
 * 处理文件上传
 */
public class UploadUtil {
    /**
     * @param file    附件
     * @param dirPath 保存的目录
     * @return
     */
    public static String saveFile(MultipartFile file, String dirPath) {
        if (Objects.isNull(file)) {
            //若file为null,则直接结束方法
            return null;
        }
        String newName = null;
        String datePath = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
        try {

            // ⽂件原名，如xxx.jpg
            String originalFilename = file.getOriginalFilename();
            // 获取⽂件的扩展名,如jpg
            String extendName =
                    originalFilename.substring(originalFilename.lastIndexOf(".") + 1, originalFilename.length());
            String uuid = UUID.randomUUID().toString();
            // 新的⽂件名字
            newName = uuid + "." + extendName;
            //创建存储的目录
            File dir = new File(dirPath);
            // 解决⽂件夹存放⽂件数量限制，按⽇期存放
            File floder = new File(dir + "/" + datePath);
            // d:/imgs/2023-03-30/
            if (!floder.exists()) {
                //若目录不存在，则创建目录
                floder.mkdirs();
            }
            //保存在磁盘文件中
            File newFile = new File(floder, newName);
            file.transferTo(newFile);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return datePath + File.separator + newName;
    }
}
